cd ../

gpus=$1
exp_name="mt_weight"

for w in 0 5 10 20 50
do
    echo "Run experiments on <${exp_name}> with ${exp_name}=${w}"
    echo "--gpus ${gpus}, --exp_name ${exp_name}/w_MT=${w}, --const_max ${w}"
    python sed_main.py --gpus ${gpus}, --exp_name ${exp_name}/w_MT=${w} --const_max ${w} --batch_sizes "[0, 12, 12, 128, 0]"
done